package main

import (
	"os"
	"log"
	"net/http"
    "github.com/cloudgo/server"
	flag "github.com/spf13/pflag"
)

// 设置 默认端口为 8080
const (
	DEFAULT string = "8080"
)

func main() {
	// 使用 默认端口
	port := os.Getenv("PORT")
	if len(port) == 0 {
		port = DEFAULT
	}

	// 用户可通过 -p 自定义端口
	_port := flag.StringP("port", "p", DEFAULT, "for httpd listening")
	flag.Parse()
	if len(*_port) != 0 {
		port = *_port
	}

	// 设置 默认访问的路由
	http.HandleFunc("/", server.Default) 

	// 得到 静态文件的路径
    // 设置 静态文件的路由
	fs := http.FileServer(http.Dir("static"))
	http.Handle("/static/",	http.StripPrefix("/static/",fs)) 
	
	//  设置 简单 js 访问的路由
	http.HandleFunc("/api/test", server.Test) 

	// 设置 提交表单并输出表格的路由
	http.HandleFunc("/login", server.Login)   

	// 设置 监听的端口
	err := http.ListenAndServe(":"+port, nil) 
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}
